package com.supergzt.aspect;


import com.supergzt.annotation.RequiresLogin;
import com.supergzt.utils.AuthUtil;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import java.lang.reflect.Method;

/**
 * @author 张钧
 * @Description
 * @create 2022-09-24 9:43
 */
@Aspect
@Component
public class PermissionCheckAspect {
	public PermissionCheckAspect() {
	}

	/**
	 * 定义AOP签名 (切入所有使用鉴权注解的方法)
	 */
	public static final String POINTCUT_SIGN =
			" @annotation(com.supergzt.annotation.RequiresLogin)";

	/**
	 * 声明AOP签名
	 */
	@Pointcut(POINTCUT_SIGN)
	public void pointcut() {
	}

	/**
	 * 环绕切入
	 *
	 * @param joinPoint 切面对象
	 * @return 底层方法执行后的返回值
	 * @throws Throwable 底层方法抛出的异常
	 */
	@Around("pointcut()")
	public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
		// 注解鉴权
		MethodSignature signature = (MethodSignature) joinPoint.getSignature();
		checkMethodAnnotation(signature.getMethod());
		// 执行原有逻辑
		return joinPoint.proceed();
	}

	/**
	 * 对一个Method对象进行注解检查
	 */
	public void checkMethodAnnotation(Method method) {
		// 校验 @RequiresLogin 注解
		RequiresLogin requiresLogin = method.getAnnotation(RequiresLogin.class);
		if (requiresLogin != null) {
			AuthUtil.checkLogin();
		}
	}
}
